package com.supconit.ai.utils.qwen;

import com.google.gson.Gson;
import com.supconit.ai.json.QWenResultObj;
import com.supconit.ai.json.Shuffle;

import java.util.HashMap;
import java.util.Map;

public class QWenResultUtils {
    /**
     * 将返回的结果转换为字符串
     * @param resultIn
     * @return
     */
    public static String convertToStr(String resultIn){
        Gson gson = new Gson();
        String msg = resultIn;
        QWenResultObj resultObj = gson.fromJson(msg, QWenResultObj.class);
        String text = resultObj.getOutput().getChoices().get(0).getMessage().getContent().get(0).getText();
        String result = Shuffle.shuffle(text);
        return result;
    }

    /**
     * 将返回的结果转换为Map
     * @param resultIn
     * @return
     */
    public static HashMap convertToMap(String resultIn){
        Gson gson = new Gson();
        String msg = resultIn;
        QWenResultObj resultObj = gson.fromJson(msg, QWenResultObj.class);
        String text = resultObj.getOutput().getChoices().get(0).getMessage().getContent().get(0).getText();
        String result = Shuffle.shuffle(text);
        HashMap map = gson.fromJson(result, HashMap.class);
        return map;
    }

}
